#pragma once
#include <iostream>
#include <sys/types.h>
#include <unistd.h>
#include <sys/wait.h>
#include <sys/stat.h>
#include <fcntl.h>

#include "../comm/log.hpp"
#include "../comm/util.hpp"

// 只负责代码编译
namespace ns_compiler
{
    using namespace ns_util;
    using namespace ns_log;

    // 引入路径拼接功能
    class Compiler
    {
    public:
        Compiler()
        {
        }

        // 返回值：编译成功：true,编译失败：false
        // file_name: hsl
        // hsl ->./temp/hsl.cpp
        // hsl ->./temp/hsl.exe
        // hsl ->./temp/hsl.stderr
        static bool Compile(const std::string &file_name)
        {
            pid_t res = fork();
            if (res < 0)
            {
                LOG(ERROR) << "内部错误，创建子进程失败" << "\n";
                return false;
            }
            else if (res == 0) // 子进程
            {
                umask(0);
                // 创建一个stderr文件，将编译错误信息重定向到该文件中
                int _stderr = open(PathUtil::CompilerError(file_name).c_str(), O_CREAT | O_WRONLY, 0644);
                if (_stderr < 0)
                {
                    LOG(WARNING) << "没有成功形成compilererror文件" << "\n";
                    exit(1);
                }
                // 重定向错误到_stderr
                dup2(_stderr, 2);

                // 子进程：调用编译器，完成对代码的编译工作
                // execlp :用于替换当前进程为一个新程序(不会创建子进程，所以要配合fork使用)
                // g++ -o target(目标) src(源文件) -std=c++11
                execlp("g++", "g++", "-o", PathUtil::Exe(file_name).c_str(), PathUtil::Src(file_name).c_str(), "-std=c++11","-D","COMPILER_ONLONE", nullptr);
                LOG(ERROR) << "启动g++编译器失败，可能是参数错误" << "\n";

                exit(2);
            }
            else
            {
                // 父进程
                waitpid(res, nullptr, 0);
                // 编译是否成功: 判断有没有生成可执行程序
                if (FileUtil::IsFileExists(PathUtil::Exe(file_name)))
                {
                    LOG(INFO) << PathUtil::Src(file_name) << " 编译成功" << "\n";
                    return true;
                }
            }
            LOG(ERROR) << "程序编译失败，未形成可执行程序" << "\n";

            return false;
        }
        ~Compiler()
        {
        }
    };

}
